Getting error when trying to calculate the length of the array after sort in python